import torch
import torch.nn as nn

class MultiLayerPerceptron(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc1 = nn.Linear(1, 3)  # 第一个全连接层，输入维度为1，输出维度为3
        self.fc2 = nn.Linear(3, 1)  # 第二个全连接层，输入维度为3，输出维度为1
        self.relu = nn.ReLU()  # 非线性激活函数

    def forward(self, x):
        x = self.fc1(x)
        x = self.relu(x)
        x = self.fc2(x)
        return x
